# 1、统计单词的数据量，将通过结果保存到文件中

# 1、读取文件
with open(file="../../data/lines.txt", encoding="utf-8", mode="r") as f:
    lines = f.readlines()

print(lines)

# 2、统计单词的数量
word_count = {}
for line in lines:
    # 去掉每行后面的换行符，再按照逗号切分
    words = line.strip().split(",")
    # 循环列表统计单词的数量
    for word in words:
        # 如果在字典中不存在存入一个1，如果在字典中存在累加一个1
        if word not in word_count:
            word_count[word] = 1
        else:
            word_count[word] += 1

print(word_count)

# 3、保存到文件中
with open(file="../../data/word_count.txt", encoding="utf-8", mode="w") as f:
    # 循环结果字典，字典中的每个元素对应文件中一行
    for word, count in word_count.items():
        f.write(f"{word},{count}\n")
